import os
from langchain_openai import ChatOpenAI

class ArkModel:
    def __init__(self):
        print(os.getenv('ARK_BASE_URL'), os.getenv('ARK_BASE_KEY'),os.getenv('ARK_MODEL_NAME'))
        self.model = ChatOpenAI(openai_api_base=os.getenv('ARK_BASE_URL'), openai_api_key=os.getenv('ARK_BASE_KEY'), model_name=os.getenv('ARK_MODEL_NAME'))

